import json

import requests

from tqdm import tqdm
from args import *
url = f"http://localhost:{port}/model/extract/summary"

def wrap(data: list):
    json_data = {
        "data": [
            {"title": i} for i in data
        ]
    }

    response = requests.post(url, json=json_data)
    return response.json()

data = [
    "俄罗斯都将发展与朝鲜的关系|外交部谈近期朝鲜半岛形势|媒体：朝鲜半岛南北对抗之弦再绷紧|朝鲜半岛军事分界线周边直面风险|垃圾气球飘动、军事协议撕毁，朝鲜半岛军事分界线周边直面风险|朝鲜称通过气球向韩投放废纸7.5吨|韩国宣布恢复对朝鲜扩音广播|朝鲜称向韩投放1400个气球7.5吨废纸|金与正：朝鲜向韩国投放7.5吨废纸|韩国向朝鲜发送美元气球|韩国宣布恢复对朝鲜扩音广播|韩国宣布将恢复对朝鲜扩音广播，哪些信息值得关注？|韩国重启对朝广播喊话 以前播过什么|韩军已启动对朝扩音广播|朝鲜再次向韩投放垃圾气球|为应对“垃圾气球”，韩军重启对朝扩音喊话|俄罗斯都将发展与朝鲜的关系|外交部谈近期朝鲜半岛形势|媒体：朝鲜半岛南北对抗之弦再绷紧|朝鲜半岛军事分界线周边直面风险|垃圾气球飘动、军事协议撕毁，朝鲜半岛军事分界线周边直面风险|朝鲜称通过气球向韩投放废纸7.5吨|韩国宣布恢复对朝鲜扩音广播|朝鲜称向韩投放1400个气球7.5吨废纸|金与正：朝鲜向韩国投放7.5吨废纸|韩国向朝鲜发送美元气球|韩国宣布恢复对朝鲜扩音广播|韩国宣布将恢复对朝鲜扩音广播，哪些信息值得关注？|韩国重启对朝广播喊话 以前播过什么|韩军已启动对朝扩音广播|朝鲜再次向韩投放垃圾气球|为应对“垃圾气球”，韩军重启对朝扩音喊话|俄罗斯都将发展与朝鲜的关系|外交部谈近期朝鲜半岛形势|媒体：朝鲜半岛南北对抗之弦再绷紧|朝鲜半岛军事分界线周边直面风险|垃圾气球飘动、军事协议撕毁，朝鲜半岛军事分界线周边直面风险|朝鲜称通过气球向韩投放废纸7.5吨|韩国宣布恢复对朝鲜扩音广播|朝鲜称向韩投放1400个气球7.5吨废纸|金与正：朝鲜向韩国投放7.5吨废纸|韩国向朝鲜发送美元气球|韩国宣布恢复对朝鲜扩音广播|韩国宣布将恢复对朝鲜扩音广播，哪些信息值得关注？|韩国重启对朝广播喊话 以前播过什么|韩军已启动对朝扩音广播|朝鲜再次向韩投放垃圾气球|为应对“垃圾气球”，韩军重启对朝扩音喊话|俄罗斯都将发展与朝鲜的关系|外交部谈近期朝鲜半岛形势"]
import time
start = time.time()
res = wrap(data)
print(json.dumps(res, ensure_ascii=False, indent=4))
print(f"cost {time.time() - start:.3f}s")